# encoding=utf8    
import numpy as np

def distance(x, y, p=2):
    '''
    input: x(ndarray):第一个样本的坐标
           y(ndarray):第二个样本的坐标
           p(int):等于1时为曼哈顿距离，等于2时为欧氏距离
    output: distance(float):x到y的距离      
    ''' 
    # ********* Begin *********#
    # 计算曼哈顿距离
    if p == 1:
        M_dis = np.sum(np.abs(x - y))
        return M_dis     

    # 计算欧氏距离
    if p == 2:
        O_dis = np.sqrt(np.sum((x - y) ** 2))
        return O_dis
    # ********* End *********#